python formula for prime numbers